home *** CD-ROM | disk | FTP | other *** search
/ Enter 2006 September / Enter 09 2006.iso / Internet / SpamExperts Home 1.1 / SpamExperts Home.exe / lib / spamexperts.modules / BmpImagePlugin.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2006-07-14  |  5.5 KB  |  175 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. __version__ = '0.7'
  5. import string
  6. import Image
  7. import ImageFile
  8. import ImagePalette
  9.  
  10. def i16(c):
  11.     return ord(c[0]) + (ord(c[1]) << 8)
  12.  
  13.  
  14. def i32(c):
  15.     return ord(c[0]) + (ord(c[1]) << 8) + (ord(c[2]) << 16) + (ord(c[3]) << 24)
  16.  
  17. BIT2MODE = {
  18.     1: ('P', 'P;1'),
  19.     4: ('P', 'P;4'),
  20.     8: ('P', 'P'),
  21.     16: ('RGB', 'BGR;16'),
  22.     24: ('RGB', 'BGR'),
  23.     32: ('RGB', 'BGRX') }
  24.  
  25. def _accept(prefix):
  26.     return prefix[:2] == 'BM'
  27.  
  28.  
  29. class BmpImageFile(ImageFile.ImageFile):
  30.     format = 'BMP'
  31.     format_description = 'Windows Bitmap'
  32.     
  33.     def _bitmap(self, header = 0, offset = 0):
  34.         if header:
  35.             self.fp.seek(header)
  36.         
  37.         read = self.fp.read
  38.         s = read(4)
  39.         s = s + ImageFile._safe_read(self.fp, i32(s) - 4)
  40.         if len(s) == 12:
  41.             bits = i16(s[10:])
  42.             self.size = (i16(s[4:]), i16(s[6:]))
  43.             compression = 0
  44.             lutsize = 3
  45.             colors = 0
  46.         elif len(s) in [
  47.             40,
  48.             64]:
  49.             bits = i16(s[14:])
  50.             self.size = (i32(s[4:]), i32(s[8:]))
  51.             compression = i32(s[16:])
  52.             lutsize = 4
  53.             colors = i32(s[32:])
  54.         else:
  55.             raise IOError('Unsupported BMP header type (%d)' % len(s))
  56.         if not colors:
  57.             colors = 1 << bits
  58.         
  59.         
  60.         try:
  61.             (self.mode, rawmode) = BIT2MODE[bits]
  62.         except KeyError:
  63.             raise IOError('Unsupported BMP pixel depth (%d)' % bits)
  64.  
  65.         if compression == 3:
  66.             mask = (i32(read(4)), i32(read(4)), i32(read(4)))
  67.             if bits == 32 and mask == (16711680, 65280, 255):
  68.                 rawmode = 'BGRX'
  69.             elif bits == 16 and mask == (63488, 2016, 31):
  70.                 rawmode = 'BGR;16'
  71.             elif bits == 16 and mask == (31744, 992, 31):
  72.                 rawmode = 'BGR;15'
  73.             else:
  74.                 raise IOError('Unsupported BMP bitfields layout')
  75.         elif compression != 0:
  76.             raise IOError('Unsupported BMP compression (%d)' % compression)
  77.         
  78.         if self.mode == 'P':
  79.             palette = []
  80.             greyscale = 1
  81.             if colors == 2:
  82.                 indices = (0, 255)
  83.             else:
  84.                 indices = range(colors)
  85.             for i in indices:
  86.                 rgb = read(lutsize)[:3]
  87.                 if rgb != chr(i) * 3:
  88.                     greyscale = 0
  89.                 
  90.                 palette.append(rgb)
  91.             
  92.             if greyscale:
  93.                 if colors == 2:
  94.                     self.mode = rawmode = '1'
  95.                 else:
  96.                     self.mode = rawmode = 'L'
  97.             else:
  98.                 self.mode = 'P'
  99.                 self.palette = ImagePalette.raw('BGR', string.join(palette, ''))
  100.         
  101.         if not offset:
  102.             offset = self.fp.tell()
  103.         
  104.         self.tile = [
  105.             ('raw', (0, 0) + self.size, offset, (rawmode, self.size[0] * bits + 31 >> 3 & -4, -1))]
  106.         self.info['compression'] = compression
  107.  
  108.     
  109.     def _open(self):
  110.         s = self.fp.read(14)
  111.         if s[:2] != 'BM':
  112.             raise SyntaxError('Not a BMP file')
  113.         
  114.         offset = i32(s[10:])
  115.         self._bitmap(offset = offset)
  116.  
  117.  
  118.  
  119. class DibImageFile(BmpImageFile):
  120.     format = 'DIB'
  121.     format_description = 'Windows Bitmap'
  122.     
  123.     def _open(self):
  124.         self._bitmap()
  125.  
  126.  
  127.  
  128. def o16(i):
  129.     return chr(i & 255) + chr(i >> 8 & 255)
  130.  
  131.  
  132. def o32(i):
  133.     return chr(i & 255) + chr(i >> 8 & 255) + chr(i >> 16 & 255) + chr(i >> 24 & 255)
  134.  
  135. SAVE = {
  136.     '1': ('1', 1, 2),
  137.     'L': ('L', 8, 256),
  138.     'P': ('P', 8, 256),
  139.     'RGB': ('BGR', 24, 0) }
  140.  
  141. def _save(im, fp, filename, check = 0):
  142.     
  143.     try:
  144.         (rawmode, bits, colors) = SAVE[im.mode]
  145.     except KeyError:
  146.         raise IOError('cannot write mode %s as BMP' % im.mode)
  147.  
  148.     if check:
  149.         return check
  150.     
  151.     stride = (im.size[0] * bits + 7) / 8 + 3 & -4
  152.     header = 40
  153.     offset = 14 + header + colors * 4
  154.     image = stride * im.size[1]
  155.     fp.write('BM' + o32(offset + image) + o32(0) + o32(offset))
  156.     fp.write(o32(header) + o32(im.size[0]) + o32(im.size[1]) + o16(1) + o16(bits) + o32(0) + o32(image) + o32(1) + o32(1) + o32(colors) + o32(colors))
  157.     fp.write('\x00' * (header - 40))
  158.     if im.mode == '1':
  159.         for i in (0, 255):
  160.             fp.write(chr(i) * 4)
  161.         
  162.     elif im.mode == 'L':
  163.         for i in range(256):
  164.             fp.write(chr(i) * 4)
  165.         
  166.     elif im.mode == 'P':
  167.         fp.write(im.im.getpalette('RGB', 'BGRX'))
  168.     
  169.     ImageFile._save(im, fp, [
  170.         ('raw', (0, 0) + im.size, 0, (rawmode, stride, -1))])
  171.  
  172. Image.register_open(BmpImageFile.format, BmpImageFile, _accept)
  173. Image.register_save(BmpImageFile.format, _save)
  174. Image.register_extension(BmpImageFile.format, '.bmp')
  175.